"""编写程序操作Excle文件，
读取文件“大家的特长.xlsx”中的内容，并统计A-H列中的内容，
在后面追加一列，对所有人的特长进行汇总。"""

from openpyxl import load_workbook

# 需要统计的数据所在的xlsx表
wb = load_workbook(r'resources\大家的特长.xlsx')
ws = wb.worksheets[0]
for index, row in enumerate(ws.rows):
    # 跳过第一行的表头
    if index == 0:
        continue
    # 获取每行的信息
    name, soccer, basketball, table_tennis, swimming, sing, dance, eat, sleep, total = map(lambda cell: cell.value, row)
    total = ''
    if soccer == '是':
        total += '足球 '
    if basketball == '是':
        total += '篮球 '
    if table_tennis == '是':
        total += '乒乓球 '
    if swimming == '是':
        total += '游泳 '
    if sing == '是':
        total += '唱歌 '
    if dance == '是':
        total += '跳舞 '
    if eat == '是':
        total += '吃饭 '
    if sleep == '是':
        total += '睡觉 '
    ws.cell(index + 1, 10, total)  # 参数分别为行、列和要导入的数据
    print(name + '的所有特长：' + total)

# 保存写入的路径，注意在程序运行时不要将xlsx文件打开
wb.save(r'resources\大家的特长.xlsx')
print('\n已成功将大家的特长汇总至表的最后一列！')
